package com.example.demo.tool.netty.chat;

import io.netty.buffer.ByteBuf;
import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.codec.ByteToMessageDecoder;
import io.netty.handler.codec.MessageToByteEncoder;

import java.util.List;

/**
 * 自定义解码器
 */
public class MyMessageDecoder extends ByteToMessageDecoder {

    /**
     * 为了处理粘包拆包
     */
    int len = 0;

    /**
     * 重写解码方法
     * @param channelHandlerContext
     * @param byteBuf
     * @param list
     * @throws Exception
     */
    @Override
    protected void decode(ChannelHandlerContext channelHandlerContext, ByteBuf byteBuf, List<Object> list) throws Exception {
        // 一个int4个字节
        if (byteBuf.readableBytes() >= 4) {
            if (len == 0) {
                len = byteBuf.readInt();
            }

            // 不是一次完整的包
            if (byteBuf.readableBytes() < len) {
                System.out.println("包不够完整");
                return;
            }

            // 组装对象
            byte[] content = new byte[len];
            byteBuf.readBytes(content);
            MyMessageProtocol protocol = new MyMessageProtocol();
            protocol.setContent(content);
            protocol.setLen(len);

            // 添加到列表，返给下一个handler处理
            list.add(protocol);
        }

        len = 0;
    }
}
